package com.spring.boot.demo.common.exception;

import com.spring.boot.demo.common.resp.Result;
import org.springframework.util.CollectionUtils;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

import java.util.List;

/**
 * @author ZhuYouBin
 * @version 1.0.0
 * @Date: 2022/10/29 21:00
 * @Description 全局异常处理类
 */
@ControllerAdvice // 指定AOP拦截Controller层方法
public class GlobalExceptionHandler {

    /**
     * 统一异常处理
     */
    @ExceptionHandler(value = {Exception.class})
    @ResponseBody
    public Result<Void> commonHandler(Exception e) {
        e.printStackTrace();
        // 返回错误信息
        return Result.failure(e.getMessage());
    }

    /**
     * 捕获我们自定义的异常
     */
    @ExceptionHandler(value = {BizException.class})
    @ResponseBody
    public Result<Void> bizExceptionHandler(BizException e) {
        e.printStackTrace();
        // 返回错误信息
        return Result.failure(e.getMessage());
    }

    /** 参数校验异常处理 */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    @ResponseBody
    public Result<Void> exception(MethodArgumentNotValidException e) {
        e.printStackTrace();
        // 保存错误提示信息
        String message = "参数校验失败";
        // 从异常对象中拿到 ObjectError 对象
        List<ObjectError> allErrors = e.getBindingResult().getAllErrors();
        if (!CollectionUtils.isEmpty(allErrors)) {
            // 返回第一个校验失败的参数名称
            message = allErrors.get(0).getDefaultMessage();
        }
        // 返回参数校验信息
        return Result.failure(message);
    }

}
